from django import template
from django.utils import timezone
import pytz

register = template.Library()


@register.filter
def local_time(value):
    if value:
        # 如果是 naive datetime，给它添加时区信息
        if timezone.is_naive(value):
            value = timezone.make_aware(value, pytz.UTC)  # 使用 pytz.UTC 来表示 UTC 时区
        return timezone.localtime(value).strftime("%Y-%m-%d %H:%M")
    return value
